package extensionfilter;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class ExtensionFilter {
    private List<String> filteredFiles = new ArrayList<>();//找到的文件放到这里

    /**
     * 递归遍历目录来找出所有特定扩展名的文件
     *
     * @param folder    要搜索的起始目录
     * @param extension 要过滤的文件扩展名
     */
    public void filterFilesByExtension(File folder, String extension){
        File[] files = folder.listFiles();
        if (files != null) {
            for (File file : files) {
                if (file.isDirectory()) {
                    filterFilesByExtension(file, extension);
                }else{
                    if (file.getName().endsWith("." + extension)) {
                        filteredFiles.add(file.getPath());
                        System.out.println("找到文件:" + file.getPath());
                    }
                }
            }
        }
    }

    public List<String> getFilteredFiles() {
        return filteredFiles;
    }
}
